package org.springframework.ai.openai.samples.helloworld.callbacks;

import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.openai.OpenAiChatModel;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Description;

import java.util.function.Function;

@Configuration
public class AIConfigs {

    @Bean
    //必须  函数功能的描述，模型使用它来选择何时以及如何调用函数。
    @Description("用来获取某地区的名字重复数量")
    public Function<LocationNamesService.Request, LocationNamesService.Response> getLocationAndNum() {
        return new LocationNamesService();
    }


    @Bean
    ChatClient chatClient(ChatClient.Builder builder) {
        return builder.defaultSystem("我叫徐庶， 我5岁。 我希望你以我爸爸的身份和我对话，你不再是ChatGpt。")
                .build();
    }


}
